/*
 * Copyright (c) Huawei Technologies Co., Ltd. 2025. All rights reserved.
 * You can use this software according to the terms and conditions of the Mulan PSL v2.
 * You may obtain a copy of Mulan PSL v2 at:
 *          http://license.coscl.org.cn/MulanPSL2
 * THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
 * EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
 * MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
 * See the Mulan PSL v2 for more details.
 */
#ifndef FLINK_TNEL_HEAPKEYEDSTATEBACKENDBUILDER_H
#define FLINK_TNEL_HEAPKEYEDSTATEBACKENDBUILDER_H
#include <emhash7.hpp>
#include "runtime/state/KeyGroupRange.h"
#include "runtime/state/HeapKeyedStateBackend.h"
#include "core/typeutils/TypeSerializer.h"

template <typename K>
class HeapKeyedStateBackendBuilder {
public:
    HeapKeyedStateBackendBuilder(TypeSerializer *keySerializer, int numberOfKeyGroups, KeyGroupRange *keyGroupRange) : keySerializer(keySerializer), numberOfKeyGroups(numberOfKeyGroups), keyGroupRange(keyGroupRange) {};
    HeapKeyedStateBackend<K> *build();
protected:
    TypeSerializer *keySerializer;
    int numberOfKeyGroups;
    KeyGroupRange *keyGroupRange;
};

template <typename K>
HeapKeyedStateBackend<K> *HeapKeyedStateBackendBuilder<K>::build()
{
    auto *keyContext = new InternalKeyContext<K>(keyGroupRange, numberOfKeyGroups);
    return new HeapKeyedStateBackend<K>(keySerializer, keyContext);
}
#endif // FLINK_TNEL_HEAPKEYEDSTATEBACKENDBUILDER_H
